/**
 * Copyright 2010-2011 Nicholas Blair, Eric Dalquist
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.googlecode.ehcache.annotations;

import java.io.Serializable;

import net.sf.ehcache.Ehcache;

import org.aopalliance.intercept.MethodInvocation;

/**
 * @author Eric Dalquist
 * @version $Revision: 656 $
 */
public class CountingTriggersRemoveInterceptor extends DefaultTriggersRemoveInterceptor {
    private int preInvRemove = 0;
    private int preInvRemoveAll = 0;
    
    @Override
    public boolean preInvokeTriggersRemove(Ehcache cache, MethodInvocation methodInvocation, Serializable key) {
        preInvRemove++;
        return super.preInvokeTriggersRemove(cache, methodInvocation, key);
    }

    @Override
    public boolean preInvokeTriggersRemoveAll(Ehcache cache, MethodInvocation methodInvocation) {
        preInvRemoveAll++;
        return super.preInvokeTriggersRemoveAll(cache, methodInvocation);
    }
    
    public void resetCounters() {
        this.preInvRemove = 0;
        this.preInvRemoveAll = 0;
    }

    public int getPreInvRemove() {
        return this.preInvRemove;
    }

    public int getPreInvRemoveAll() {
        return this.preInvRemoveAll;
    }

    @Override
    public String toString() {
        return "CountingTriggersRemoveInterceptor [preInvRemove=" + this.preInvRemove + ", preInvRemoveAll="
                + this.preInvRemoveAll + "]";
    }
}
